"""
定义类 和 方法
"""
import requests
import app


class IhrmApi(object):
    def __init__(self):
        self.url = app.BASE_URL + "/api/sys/login"
        self.headers = app.HEADERS

    # 登录方法
    def login(self, json_data):
        resp = requests.post(url=self.url, headers=self.headers, json=json_data)
        return resp

    # 添加员工
    def add_emp(self, headers, req_data):
        resp = requests.post(url=app.BASE_URL + "/api/sys/user",
                             headers=headers, json=req_data)
        return resp

    # 查询员工
    def query_emp(self, emp_id, headers):
        return requests.get(url=app.BASE_URL + "/api/sys/user/" + emp_id, headers=headers)

    # 修改员工
    def modify_emp(self, emp_id, headers, req_data):
        return requests.put(url=app.BASE_URL + "/api/sys/user/" + emp_id, headers=headers, json=req_data)

    # 删除员工
    def delete_emp(self, emp_id, headers):
        return requests.delete(url=app.BASE_URL + "/api/sys/user/" + emp_id, headers=headers)

    # 查询员工列表
    def query_emp_list(self, headers):
        return requests.get(url=app.BASE_URL+"/api/sys/user?page=1&size=10", headers=headers)
